1786F - Wooden Spoon - CodeForces Solution


combinatorics dp fft games math *2400

Please click on ads to support us..

C++ Code:

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL Mod = 998244353;
void Solve() {
    int n; cin >> n;
    int N = (1 << n);
    vector<LL> fac(N + 1), inv(N + 1), ifac(N + 1);
    fac[0] = fac[1] = inv[1] = ifac[0] = ifac[1] = 1;
    for (int i = 2; i <= N; ++i) {
        fac[i] = fac[i - 1] * i % Mod;
        inv[i] = Mod - Mod / i * inv[Mod % i] % Mod;
        ifac[i] = ifac[i - 1] * inv[i] % Mod;
    }
    function<LL(int, int)> C = [&](int m, int n) {
        if (n < 0 || m < n) return 0ll;
        return fac[m] * ifac[n] % Mod * ifac[m - n] % Mod;
    };
 
    vector<LL> f(N + 1), sum(N + 1);
    f[1] = (N * fac[1 << n - 1]) % Mod;
    for (int i = n - 2; i >= 0; --i) {
        for (int j = 1; j <= N; ++j) sum[j] = (sum[j - 1] + f[j]) % Mod;
        for (int j = 1; j <= N; ++j) {
            f[j] = sum[j - 1] * C(N - j - (1 << i), (1 << i) - 1) % Mod;
            f[j] = f[j] * fac[1 << i] % Mod;
        }
    }
    LL ans = 0;
    for (int i = 1; i <= N; ++i) {
        cout << ans << '\n';
        ans = (ans + f[i]) % Mod;
    }
}
int main() {
    ios::sync_with_stdio(false); cin.tie(nullptr);
    int t = 1;
    // cin >> t;
    while (t--) Solve();
    return 0;
}


Comments

Submit
0 Comments
More Questions

866. Prime Palindrome
1516A - Tit for Tat
622. Design Circular Queue
814. Binary Tree Pruning
791. Custom Sort String
787. Cheapest Flights Within K Stops
779. K-th Symbol in Grammar
701. Insert into a Binary Search Tree
429. N-ary Tree Level Order Traversal
739. Daily Temperatures
647. Palindromic Substrings
583. Delete Operation for Two Strings
518. Coin Change 2
516. Longest Palindromic Subsequence
468. Validate IP Address
450. Delete Node in a BST
445. Add Two Numbers II
442. Find All Duplicates in an Array
437. Path Sum III
436. Find Right Interval
435. Non-overlapping Intervals
406. Queue Reconstruction by Height
380. Insert Delete GetRandom O(1)
332. Reconstruct Itinerary
368. Largest Divisible Subset
377. Combination Sum IV
322. Coin Change
307. Range Sum Query - Mutable
287. Find the Duplicate Number
279. Perfect Squares